package com.gpStudy.Netty.rpc.consumer.proxy;

import com.gpStudy.Netty.rpc.protocol.InvokerProtocol;
import com.gpStudy.Netty.rpc.registry.RegistryHandler;
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.LengthFieldBasedFrameDecoder;
import io.netty.handler.codec.LengthFieldPrepender;
import io.netty.handler.codec.serialization.ClassResolvers;
import io.netty.handler.codec.serialization.ObjectDecoder;
import io.netty.handler.codec.serialization.ObjectEncoder;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;

public class RpcProxy {

    public static <T> T create(Class<?> clazz){
        MethodProxy proxy = new MethodProxy(clazz);
        Class<?> [] interfaces = clazz.isInterface() ?
                                new Class[]{clazz} :
                                clazz.getInterfaces();
        T result = (T) Proxy.newProxyInstance(clazz.getClassLoader(),interfaces,proxy);
        return result;
    }

    // 将 本地调用，通过代理的形式转变成网络调用
    private static class MethodProxy implements InvocationHandler{

        private Class<?> clazz;

        public MethodProxy(Class<?> clazz) {
            this.clazz = clazz;
        }

        @Override
        public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

            if (Object.class.equals(method.getDeclaringClass())){// 是否是一个实现类
                return method.invoke(this, args);
            }else{// 接口
                return rpcInvoker(proxy, method, args);
            }
        }

        private Object rpcInvoker(Object proxy, Method method, Object[] args) {
            // 先要构造一个协议内容，消息
            InvokerProtocol msg = new InvokerProtocol();
            msg.setClassName(this.clazz.getName());
            msg.setMethodName(method.getName());
            msg.setParams(method.getParameterTypes());
            msg.setValues(args);

            // 发起网络请求
            EventLoopGroup workgroup = new NioEventLoopGroup();

            final RpcProxyHandler rpcProxyHandler = new RpcProxyHandler();

            Bootstrap client = new Bootstrap();
            try {
                client.group(workgroup)
                        .channel(NioSocketChannel.class)
                        .option(ChannelOption.TCP_NODELAY, true)
                        .handler(new ChannelInitializer<SocketChannel>() {
                            @Override
                            protected void initChannel(SocketChannel socketChannel) throws Exception {
                                ChannelPipeline pipeline = socketChannel.pipeline();

                                // 就是对我们处理逻辑的封装
                                // 对于自定义协议的内容进行编、解码

                                //自定义协议解码器
								/** 入参有5个，分别解释如下
								 maxFrameLength：框架的最大长度。如果帧的长度大于此值，则将抛出TooLongFrameException。
								 lengthFieldOffset：长度字段的偏移量：即对应的长度字段在整个消息数据中得位置
								 lengthFieldLength：长度字段的长度：如：长度字段是int型表示，那么这个值就是4（long型就是8）
								 lengthAdjustment：要添加到长度字段值的补偿值
								 initialBytesToStrip：从解码帧中去除的第一个字节数
								 */
                                pipeline.addLast( new LengthFieldBasedFrameDecoder(Integer.MAX_VALUE, 0,4,0,4));

                                pipeline.addLast(new LengthFieldPrepender(4));

                                //对象参数类型编码器

                                // 实参处理
                                pipeline.addLast("encoder", new ObjectEncoder());

                                pipeline.addLast("decoder", new ObjectDecoder(Integer.MAX_VALUE, ClassResolvers.cacheDisabled(null)));

                                // 前面的编解码，就是完成对数据的解析
                                // 最后一步，执行属于自己的逻辑
                                // 1、注册，给每个对象起一个名字，对外提供服务的名字
                                // 2、服务要做一个登记
                                pipeline.addLast("handler",rpcProxyHandler);
                            }
                        });
                ChannelFuture future = client.connect("localhost", 8080).sync();
                future.channel().writeAndFlush(msg).sync();
                future.channel().closeFuture().sync();

            } catch (InterruptedException e) {
                e.printStackTrace();
            }finally {
                workgroup.shutdownGracefully();
            }
            return rpcProxyHandler.getResponse();


        }
    }
}
